06. 1x1 Convolution Quiz Solution
The correct use is
tf.layers.conv2d(x, num_outputs, 1, 1, weights_initializer=custom_init)
.
-
num_outputs
defines the number of output channels or kernels - The third argument is the kernel size, which is 1.
- The fourth argument is the stride, we set this to 1.
- We use the custom initializer so the weights in the dense and convolutional layers are identical.
This results in the a matrix multiplication operation that preserves spatial information.